tensor([[0.4800, 0.4800, 0.4800, 0.4800, 0.4800],
        [0.5200, 0.5200, 0.5200, 0.5200, 0.5200]], grad_fn=<RoundBackward1>)